python program square of a number